﻿using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Data;
using System.Data.Common;
using System.Data.SQLite;

namespace KD.Sms
{
    public class SQLiteHelper
    {
        //读取数据库连接配置
        public static string ConnLocalString = @"data source=" + System.Windows.Forms.Application.StartupPath + "\\KD.DataBase.db;";

        public static bool ExecSQLTrans(List<string> list)
        {
            try
            {
                using (SQLiteConnection conn = new SQLiteConnection(ConnLocalString))
                {
                    conn.Open();
                    SQLiteTransaction trans = conn.BeginTransaction();

                    foreach (string _str in list)
                    {
                        ExecuteNonQuery(trans, CommandType.Text, _str, null);
                    }
                    trans.Commit();
                }
                return true;
            }
            catch (Exception ex)
            {
                return false;
            }
        }

        /// <summary>
        /// 获取Dataset数据集合
        /// </summary>
        /// <param name="str_sql">SQL语句</param>
        /// <returns></returns>
        public static DataSet GetDataSet(string str_sql)
        {
            DataSet ds = new DataSet();
            using (SQLiteConnection conn = new SQLiteConnection(ConnLocalString))
            {
                conn.Open();
                SQLiteDataAdapter da = new SQLiteDataAdapter(str_sql, conn);

                da.Fill(ds);

                if (ds != null && ds.Tables.Count > 0)
                    return ds;
                else
                    return null;
            }
        }


        /// <summary>
        /// 获取Dataset数据集合
        /// </summary>
        /// <param name="str_sql">SQL语句</param>
        /// <returns></returns>
        public static DataTable GetDataTable(string str_sql)
        {
            DataSet ds = new DataSet();
            using (SQLiteConnection conn = new SQLiteConnection(ConnLocalString))
            {
                conn.Open();
                SQLiteDataAdapter da = new SQLiteDataAdapter(str_sql, conn);

                da.Fill(ds);

                if (ds != null && ds.Tables.Count > 0)
                    return ds.Tables[0];
                else
                    return null;
            }
        }

        /// <summary>
        /// 获取Dataset数据集合
        /// </summary>
        /// <param name="str_sql">SQL语句</param>
        /// <returns></returns>
        public static DataTable GetDataTable(string connStr, string str_sql)
        {
            DataSet ds = new DataSet();
            using (SQLiteConnection conn = new SQLiteConnection(connStr))
            {
                conn.Open();
                SQLiteDataAdapter da = new SQLiteDataAdapter(str_sql, conn);

                da.Fill(ds);

                if (ds != null && ds.Tables.Count > 0)
                    return ds.Tables[0];
                else
                    return null;
            }
        }

        /// <summary>
        /// ExecuteQuery
        /// </summary>
        /// <param name="strSql"></param>
        /// <param name="commandParameters"></param>
        /// <returns></returns>
        public static DataTable ExecuteQuery(string strSql, params SQLiteParameter[] commandParameters)
        {
            return ExecuteQuery(ConnLocalString, CommandType.Text, strSql, commandParameters);
        }

        /// <summary>
        /// ExecuteQuery
        /// </summary>
        /// <param name="ConnLocalString"></param>
        /// <param name="cmdType"></param>
        /// <param name="cmdText"></param>
        /// <param name="commandParameters"></param>
        /// <returns></returns>
        public static DataTable ExecuteQuery(string ConnLocalString, string cmdText, params SQLiteParameter[] commandParameters)
        {
            // Create a new Oracle command
            SQLiteCommand cmd = new SQLiteCommand();
            DataSet ds = new DataSet();
            //Create a connection
            using (SQLiteConnection connection = new SQLiteConnection(ConnLocalString))
            {
                //Prepare the command
                PrepareCommand(cmd, connection, null, CommandType.Text, cmdText, commandParameters);
                //Execute the command
                SQLiteDataAdapter da = new SQLiteDataAdapter(cmd);
                da.Fill(ds);

                if (ds != null && ds.Tables.Count > 0)
                    return ds.Tables[0];
                else
                    return null;
            }

        }


        /// <summary>
        /// ExecuteQuery
        /// </summary>
        /// <param name="ConnLocalString"></param>
        /// <param name="cmdType"></param>
        /// <param name="cmdText"></param>
        /// <param name="commandParameters"></param>
        /// <returns></returns>
        public static DataTable ExecuteQuery(string ConnLocalString, CommandType cmdType, string cmdText, params SQLiteParameter[] commandParameters)
        {
            // Create a new Oracle command
            SQLiteCommand cmd = new SQLiteCommand();
            DataSet ds = new DataSet();
            //Create a connection
            using (SQLiteConnection connection = new SQLiteConnection(ConnLocalString))
            {
                //Prepare the command
                PrepareCommand(cmd, connection, null, cmdType, cmdText, commandParameters);
                //Execute the command
                SQLiteDataAdapter da = new SQLiteDataAdapter(cmd);
                da.Fill(ds);

                if (ds != null && ds.Tables.Count > 0)
                    return ds.Tables[0];
                else
                    return null;
            }

        }

        //Create a hashtable for the parameter cached
        private static Hashtable parmCache = Hashtable.Synchronized(new Hashtable());


        /// <summary>
        /// Execute a database query which does not include a select
        /// </summary>
        /// <param name="connString">Connection string to database</param>
        /// <param name="cmdType">Command type either stored procedure or SQL</param>
        /// <param name="cmdText">Acutall SQL Command</param>
        /// <param name="commandParameters">Parameters to bind to the command</param>
        /// <returns></returns>
        public static decimal ExecuteNonQuery(string ConnLocalString, CommandType cmdType, string cmdText, params SQLiteParameter[] commandParameters)
        {
            // Create a new Oracle command
            SQLiteCommand cmd = new SQLiteCommand();
            //Create a connection
            using (SQLiteConnection connection = new SQLiteConnection(ConnLocalString))
            {
                //Prepare the command
                PrepareCommand(cmd, connection, null, cmdType, cmdText, commandParameters);
                //Execute the command
                decimal val = cmd.ExecuteNonQuery();
                cmd.Parameters.Clear();
                return val;
            }
        }

        /// <summary>
        /// 测试数据库是否能连接。

        /// 
        /// </summary>
        /// <returns>Ture。连接成功</returns>
        public static bool CanConnection()
        {
            bool ret = false;
            try
            {
                using (SQLiteConnection connection = new SQLiteConnection(ConnLocalString))
                {
                    connection.Open();
                    ret = true;
                    connection.Close();
                    connection.Dispose();
                }
            }
            catch
            {

            }
            return ret;
        }

        /// <summary>
        /// Execute a database query which does not include a select
        /// </summary>
        /// <param name="connString">Connection string to database</param>
        /// <param name="cmdType">Command type either stored procedure or SQL</param>
        /// <param name="cmdText">Acutall SQL Command</param>
        /// <param name="commandParameters">Parameters to bind to the command</param>
        /// <returns></returns>
        public static decimal ExecuteNonQuery(string cmdText, params SQLiteParameter[] commandParameters)
        {
            // Create a new Oracle command
            SQLiteCommand cmd = new SQLiteCommand();
            //Create a connection
            using (SQLiteConnection connection = new SQLiteConnection(ConnLocalString))
            {
                //Prepare the command
                PrepareCommand(cmd, connection, null, CommandType.Text, cmdText, commandParameters);
                //Execute the command
                decimal val = cmd.ExecuteNonQuery();
                cmd.Parameters.Clear();
                return val;
            }
        }


        /// <summary>
        /// Execute an SQLiteCommand (that returns no resultset) against an existing database transaction 
        /// using the provided parameters.
        /// </summary>
        /// <remarks>
        /// e.g.:  
        ///  decimal result = ExecuteNonQuery(trans, CommandType.StoredProcedure, "PublishOrders", new SQLiteParameter(":prodid", 24));
        /// </remarks>
        /// <param name="trans">an existing database transaction</param>
        /// <param name="commandType">the CommandType (stored procedure, text, etc.)</param>
        /// <param name="commandText">the stored procedure name or PL/SQL command</param>
        /// <param name="commandParameters">an array of OracleParamters used to execute the command</param>
        /// <returns>an decimal representing the number of rows affected by the command</returns>
        public static decimal ExecuteNonQuery(SQLiteTransaction trans, CommandType cmdType, string cmdText, params SQLiteParameter[] commandParameters)
        {
            SQLiteCommand cmd = new SQLiteCommand();
            PrepareCommand(cmd, trans.Connection, trans, cmdType, cmdText, commandParameters);
            decimal val = cmd.ExecuteNonQuery();
            cmd.Parameters.Clear();
            return val;
        }

        /// <summary>
        /// Execute an SQLiteCommand (that returns no resultset) against an existing database connection 
        /// using the provided parameters.
        /// </summary>
        /// <remarks>
        /// e.g.:  
        ///  decimal result = ExecuteNonQuery(connString, CommandType.StoredProcedure, "PublishOrders", new SQLiteParameter(":prodid", 24));
        /// </remarks>
        /// <param name="conn">an existing database connection</param>
        /// <param name="commandType">the CommandType (stored procedure, text, etc.)</param>
        /// <param name="commandText">the stored procedure name or PL/SQL command</param>
        /// <param name="commandParameters">an array of OracleParamters used to execute the command</param>
        /// <returns>an decimal representing the number of rows affected by the command</returns>
        public static decimal ExecuteNonQuery(SQLiteConnection connection, CommandType cmdType, string cmdText, params SQLiteParameter[] commandParameters)
        {
            SQLiteCommand cmd = new SQLiteCommand();
            PrepareCommand(cmd, connection, null, cmdType, cmdText, commandParameters);
            decimal val = cmd.ExecuteNonQuery();
            cmd.Parameters.Clear();
            return val;
        }

        /// <summary>
        /// Execute a select query that will return a result set
        /// </summary>
        /// <param name="connString">Connection string</param>
        //// <param name="commandType">the CommandType (stored procedure, text, etc.)</param>
        /// <param name="commandText">the stored procedure name or PL/SQL command</param>
        /// <param name="commandParameters">an array of OracleParamters used to execute the command</param>
        /// <returns></returns>
        public static SQLiteDataReader ExecuteReader(string ConnLocalString, CommandType cmdType, string cmdText, params SQLiteParameter[] commandParameters)
        {
            //Create the command and connection
            SQLiteCommand cmd = new SQLiteCommand();
            SQLiteConnection conn = new SQLiteConnection(ConnLocalString);
            try
            {
                //Prepare the command to execute
                PrepareCommand(cmd, conn, null, cmdType, cmdText, commandParameters);

                //Execute the query, stating that the connection should close when the resulting datareader has been read
                SQLiteDataReader rdr = cmd.ExecuteReader(CommandBehavior.CloseConnection);
                cmd.Parameters.Clear();
                return rdr;

            }
            catch
            {

                //If an error occurs close the connection as the reader will not be used and we expect it to close the connection
                conn.Close();
                throw;
            }
        }

        /// <summary>
        /// Execute an SQLiteCommand that returns the first column of the first record against the database specified in the connection string 
        /// using the provided parameters.
        /// </summary>
        /// <remarks>
        /// e.g.:  
        ///  Object obj = ExecuteScalar(connString, CommandType.StoredProcedure, "PublishOrders", new SQLiteParameter(":prodid", 24));
        /// </remarks>
        /// <param name="ConnLocalString">a valid connection string for a SQLiteConnection</param>
        /// <param name="commandType">the CommandType (stored procedure, text, etc.)</param>
        /// <param name="commandText">the stored procedure name or PL/SQL command</param>
        /// <param name="commandParameters">an array of OracleParamters used to execute the command</param>
        /// <returns>An object that should be converted to the expected type using Convert.To{Type}</returns>
        public static object ExecuteScalar(string cmdText, params SQLiteParameter[] commandParameters)
        {
            // Create a new Oracle command
            SQLiteCommand cmd = new SQLiteCommand();
            //Create a connection
            using (SQLiteConnection connection = new SQLiteConnection(ConnLocalString))
            {
                PrepareCommand(cmd, connection, null, CommandType.Text, cmdText, commandParameters);
                object val = cmd.ExecuteScalar();
                cmd.Parameters.Clear();
                return val;
            }
        }


        /// <summary>
        /// Execute an SQLiteCommand that returns the first column of the first record against the database specified in the connection string 
        /// using the provided parameters.
        /// </summary>
        /// <remarks>
        /// e.g.:  
        ///  Object obj = ExecuteScalar(connString, CommandType.StoredProcedure, "PublishOrders", new SQLiteParameter(":prodid", 24));
        /// </remarks>
        /// <param name="ConnLocalString">a valid connection string for a SQLiteConnection</param>
        /// <param name="commandType">the CommandType (stored procedure, text, etc.)</param>
        /// <param name="commandText">the stored procedure name or PL/SQL command</param>
        /// <param name="commandParameters">an array of OracleParamters used to execute the command</param>
        /// <returns>An object that should be converted to the expected type using Convert.To{Type}</returns>
        public static object ExecuteScalar(string ConnLocalString, CommandType cmdType, string cmdText, params SQLiteParameter[] commandParameters)
        {
            SQLiteCommand cmd = new SQLiteCommand();
            using (SQLiteConnection conn = new SQLiteConnection(ConnLocalString))
            {
                PrepareCommand(cmd, conn, null, cmdType, cmdText, commandParameters);
                object val = cmd.ExecuteScalar();
                cmd.Parameters.Clear();
                return val;
            }
        }

        ///	<summary>
        ///	Execute	a SQLiteCommand (that returns a 1x1 resultset)	against	the	specified SQLiteTransaction
        ///	using the provided parameters.
        ///	</summary>
        ///	<param name="transaction">A	valid SQLiteTransaction</param>
        ///	<param name="commandType">The CommandType (stored procedure, text, etc.)</param>
        ///	<param name="commandText">The stored procedure name	or PL/SQL command</param>
        ///	<param name="commandParameters">An array of	OracleParamters used to execute the command</param>
        ///	<returns>An	object containing the value	in the 1x1 resultset generated by the command</returns>
        public static object ExecuteScalar(SQLiteTransaction transaction, CommandType commandType, string commandText, params SQLiteParameter[] commandParameters)
        {
            if (transaction == null)
                throw new ArgumentNullException("transaction");
            if (transaction != null && transaction.Connection == null)
                throw new ArgumentException("The transaction was rollbacked	or commited, please	provide	an open	transaction.", "transaction");
            // Create a	command	and	prepare	it for execution
            SQLiteCommand cmd = new SQLiteCommand();
            PrepareCommand(cmd, transaction.Connection, transaction, commandType, commandText, commandParameters);
            // Execute the command & return	the	results
            object retval = cmd.ExecuteScalar();

            // Detach the SqlParameters	from the command object, so	they can be	used again
            cmd.Parameters.Clear();
            return retval;
        }

        /// <summary>
        /// Execute an SQLiteCommand that returns the first column of the first record against an existing database connection 
        /// using the provided parameters.
        /// </summary>
        /// <remarks>
        /// e.g.:  
        ///  Object obj = ExecuteScalar(conn, CommandType.StoredProcedure, "PublishOrders", new SQLiteParameter(":prodid", 24));
        /// </remarks>
        /// <param name="conn">an existing database connection</param>
        /// <param name="commandType">the CommandType (stored procedure, text, etc.)</param>
        /// <param name="commandText">the stored procedure name or PL/SQL command</param>
        /// <param name="commandParameters">an array of OracleParamters used to execute the command</param>
        /// <returns>An object that should be converted to the expected type using Convert.To{Type}</returns>
        public static object ExecuteScalar(SQLiteConnection ConnLocalString, CommandType cmdType, string cmdText, params SQLiteParameter[] commandParameters)
        {
            SQLiteCommand cmd = new SQLiteCommand();
            PrepareCommand(cmd, ConnLocalString, null, cmdType, cmdText, commandParameters);
            object val = cmd.ExecuteScalar();
            cmd.Parameters.Clear();
            return val;
        }

        /// <summary>
        /// Add a set of parameters to the cached
        /// </summary>
        /// <param name="cacheKey">Key value to look up the parameters</param>
        /// <param name="commandParameters">Actual parameters to cached</param>
        public static void CacheParameters(string cacheKey, params SQLiteParameter[] commandParameters)
        {
            parmCache[cacheKey] = commandParameters;
        }

        /// <summary>
        /// Fetch parameters from the cache
        /// </summary>
        /// <param name="cacheKey">Key to look up the parameters</param>
        /// <returns></returns>
        public static SQLiteParameter[] GetCachedParameters(string cacheKey)
        {
            SQLiteParameter[] cachedParms = (SQLiteParameter[])parmCache[cacheKey];

            if (cachedParms == null)
                return null;

            // If the parameters are in the cache
            SQLiteParameter[] clonedParms = new SQLiteParameter[cachedParms.Length];

            // return a copy of the parameters
            for (int i = 0, j = cachedParms.Length; i < j; i++)
                clonedParms[i] = (SQLiteParameter)((ICloneable)cachedParms[i]).Clone();

            return clonedParms;
        }

        /// <summary>
        /// internal function to prepare a command for execution by the database
        /// </summary>
        /// <param name="cmd">Existing command object</param>
        /// <param name="conn">Database connection object</param>
        /// <param name="trans">Optional transaction object</param>
        /// <param name="cmdType">Command type, e.g. stored procedure</param>
        /// <param name="cmdText">Command test</param>
        /// <param name="commandParameters">Parameters for the command</param>
        private static void PrepareCommand(SQLiteCommand cmd, SQLiteConnection conn, SQLiteTransaction trans, CommandType cmdType, string cmdText, SQLiteParameter[] commandParameters)
        {

            //Open the connection if required
            if (conn.State != ConnectionState.Open)
                conn.Open();

            //Set up the command
            cmd.Connection = conn;
            cmd.CommandText = cmdText;
            cmd.CommandType = cmdType;

            //Bind it to the transaction if it exists
            if (trans != null)
                cmd.Transaction = trans;

            // Bind the parameters passed in
            if (commandParameters != null)
            {
                foreach (SQLiteParameter parm in commandParameters)
                    cmd.Parameters.Add(parm);
            }
        }

        /// <summary>
        /// Converter to use boolean data type with Oracle
        /// </summary>
        /// <param name="value">Value to convert</param>
        /// <returns></returns>
        public static string OraBit(bool value)
        {
            if (value)
                return "Y";
            else
                return "N";
        }

        /// <summary>
        /// Converter to use boolean data type with Oracle
        /// </summary>
        /// <param name="value">Value to convert</param>
        /// <returns></returns>
        public static bool OraBool(string value)
        {
            if (value.Equals("Y"))
                return true;
            else
                return false;
        }
    }
}
